#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Author: CoderXYX
# @Email : xue752834620@126.com
# @Date  : 2019/5/3
# @Software: PyCharm

# 将一张图片填充为正方形后切为9块
from PIL import Image
import sys


# 将图片填充为正方形
def fill_image(image_fill):
    width, height = image.size
    # 选取长和宽中较大值作为新图片的长宽
    new_image_length = width if width > height else height
    # 生成新图片[白底]
    new_image = Image.new(image_fill.mode, (new_image_length, new_image_length), color='white')
    # 将之前的图片粘贴在新图上，且居中
    if width > height:  # 原图宽大于高，则填充图片的竖直维度
        # (x,y)二元组表示粘贴上图相对下图的起始位置
        new_image.paste(image_fill, (0, int((new_image_length - height) / 2)))
    else:
        new_image.paste(image_fill, (int((new_image_length - width) / 2), 0))
    return new_image


# 切图
def cut_image(image_cut):
    width, height = image_cut.size
    item_width = int(width / 3)
    box_list = []
    # (left, upper, right, lower)
    for i in range(0, 3):  # 两重循环，生成9张图片基于原图的位置        
        for j in range(0, 3):
            # print((i*item_width,j*item_width,(i+1)*item_width,(j+1)*item_width))
            box = (j * item_width, i * item_width, (j + 1) * item_width, (i + 1) * item_width)
            box_list.append(box)
    cut_image_list = [image_cut.crop(box) for box in box_list]
    return cut_image_list


# 保存
def save_images(save_image_list):
    index = 1
    for image_save in save_image_list:
        # 保存到指定位置
        # image_save.save("E:\\Code\\Python\\Xdoit\\com\\Resources" + image_name, 'PNG')
        # 保存到根目录下(相对路径)
        # image_save.save('./Result/python' + str(index) + '.png', 'PNG')
        # 保存到根目录下(绝对路径)
        image_save.save('Result\\python' + str(index) + '.png', 'PNG')
        index += 1


if __name__ == '__main__':
    # 相对路径
    # file_path = "./Resources/python.png"
    # 绝对路径
    file_path = "Resources\\python.png"
    # 从根目录打开
    image = Image.open(file_path)
    # image.show()
    image = fill_image(image)
    image_list = cut_image(image)
    save_images(image_list)
